from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
import time
# from msedge.selenium_tools import Edge, EdgeOptions
from selenium.webdriver.edge.options import Options


def test():
    driver = webdriver.Edge()
    driver.get("https://www.baidu.com")
    assert u"百度" in driver.title
    # 获取元素
    elem = driver.find_element(By.ID, 'kw')
    # 清除元素内容
    elem.clear()
    # 元素中填充内容
    elem.send_keys(u"网络爬虫")
    # 输入回车键，开始搜索
    elem.send_keys(Keys.RETURN)
    # 如果获取的是button按钮，使用click方法，模拟点击事件
    time.sleep(3)
    assert u"网络爬虫." not in driver.page_source
    driver.close()


def no_header_test():
    edge_options = Options()
    # 使用无头模式
    edge_options.add_argument('--headless')
    # 禁用GPU，防止无头模式出现莫名的BUG
    edge_options.add_argument('--disable-gpu')
    # 将参数传给浏览器
    browser = webdriver.Edge(options=edge_options)
    # 启动浏览器
    url = "https://baidu.com"
    browser.get(url)
    print(browser.title)
    browser.quit()
